package com.filter;

import java.io.IOException;

import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;

import org.hibernate.HibernateException;
import org.hibernate.Session;
import org.hibernate.cfg.Configuration;
import org.hibernate.service.ServiceRegistry;
import org.hibernate.service.ServiceRegistryBuilder;

@SuppressWarnings("deprecation")
public class HibernateSessionFilter implements Filter {
	private static String CONFIG_FILE_LOCATION = "/hibernate.cfg.xml";
	private static final ThreadLocal<Session> threadLocal = new ThreadLocal<Session>();

	private static Configuration configuration = new Configuration();
	private static org.hibernate.SessionFactory sessionFactory;
	private static String configFile = CONFIG_FILE_LOCATION;
	private static ServiceRegistry serviceRegistry = null;

	public void init(FilterConfig arg0) throws ServletException {
		try {
			configuration.configure(configFile);
			serviceRegistry = new ServiceRegistryBuilder().applySettings(
					configuration.getProperties()).buildServiceRegistry();

			sessionFactory = configuration.buildSessionFactory(serviceRegistry);



		} catch (Exception e) {
			System.err.println("%%%% Error Creating SessionFactory %%%%");
			e.printStackTrace();
		}

	}

	public static void rebuildSessionFactory() {
		try {
			configuration.configure(configFile);
			serviceRegistry = new ServiceRegistryBuilder().applySettings(
					configuration.getProperties()).buildServiceRegistry();

			sessionFactory = configuration.buildSessionFactory(serviceRegistry);
		} catch (Exception e) {
			System.err.println("%%%% Error Creating SessionFactory %%%%");
			e.printStackTrace();
		}
	}

	public static Session getSession() throws HibernateException {
		Session session = (Session) threadLocal.get();

		if (session == null || !session.isOpen()) {
			if (sessionFactory == null) {
				rebuildSessionFactory();
			}
			session = (sessionFactory != null) ? sessionFactory.openSession()
					: null;
			threadLocal.set(session);
		}
		return session;
	}
	
	public static void closeSession() throws HibernateException {

		Session session = (Session) threadLocal.get();
		threadLocal.set(null);

		if (session != null) {
			System.out.println("YYYYY");
			session.close();
		}
	}
	

	public static org.hibernate.SessionFactory getSessionFactory() {
		return sessionFactory;
	}

	public void doFilter(ServletRequest req, ServletResponse res,
			FilterChain chain) throws IOException, ServletException {
		try {
			System.out.println("Request...");
			chain.doFilter(req, res);	
		} catch (Exception e) {
			e.printStackTrace();
		}finally{
			System.out.println("Response...");
			
			HibernateSessionFilter.closeSession();
		}
	}

	public void destroy() {

	}

}
